package com.lin.netty.module.handler;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.buffer.ByteBuf;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.ServerSocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.string.StringDecoder;
import lombok.extern.slf4j.Slf4j;

import java.nio.ByteBuffer;

/**
 * ChannelHandler 用来处理 Channel 上的各种事件，分为入站、出站两种。所有 ChannelHandler 被连成一串，就是 Pipeline
 * <p>
 * * 入站处理器通常是 ChannelInboundHandlerAdapter 的子类，主要用来读取客户端数据，处理客户端的入站消息
 * SimpleChannelInboundHandler<消息的类型>，当收到符合消息类型的消息时，会触发channelRead0方法来处理消息。
 * * 出站处理器通常是 ChannelOutboundHandlerAdapter 的子类，主要对服务端写回结果进行加工
 * <p>
 * 打个比喻，每个 Channel 是一个产品的加工车间，Pipeline 是车间中的流水线，ChannelHandler 就是流水线上的各道工序，而后面要讲的 ByteBuf 是原材料，经过很多工序的加工：先经过一道道入站工序，再经过一道道出站工序最终变成产品
 */
@Slf4j
public class InOutBoundHandlerTest {
    public static void main(String[] args) {
        ServerBootstrap serverBootstrap = new ServerBootstrap();
        NioEventLoopGroup boss = new NioEventLoopGroup();
        NioEventLoopGroup workers = new NioEventLoopGroup();
        serverBootstrap
                .group(boss, workers)
                .channel(NioServerSocketChannel.class)
                .childHandler(new ChannelInitializer<NioSocketChannel>() {
                    @Override
                    protected void initChannel(NioSocketChannel ch) throws Exception {
                        log.debug("收到客户端连接：{}", ch);
                        ChannelPipeline pipeline = ch.pipeline();
                        /**
                         * 添加handler handler双向链表：
                         * head -> stringDecoder -> h1 -> h2 -> h3 -> o1 -> o2 -> o3 -> tail
                         * 消息入站的话从前往后执行(h1,h2,h3)，消息写回出站的话从后往前执行(o3,o2,o1)
                         *
                         * 如果要想解耦既每个handler对消息进行不同的加工处理的话，可以使用ctx.fireChannelRead(msg);将消息传给下一个handler处理。
                         * 如果没有使用 ctx.fireChannelRead(msg); 的话，消息入站后，由第一个符合消息类型的handler处理既h1
                         *
                         * StringDecoder也是一个入站处理器，将收到的消息ByteBuf解码成一个字符串，
                         * 然后调用ctx.fireChannelRead(out.getUnsafe(i));传给下一个handler
                         */
                        // 添加入站处理器
                        pipeline.addLast("stringDecoder", new StringDecoder());

                        pipeline.addLast("h1", new SimpleChannelInboundHandler<String>() {
                            @Override
                            protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
                                log.debug("h1 收到的消息：{}", msg);
                                msg += "lll";
                                ctx.fireChannelRead(msg);// 调用并传递消息给下一个handler
                            }
                        });

                        pipeline.addLast("h2", new SimpleChannelInboundHandler<String>() {
                            @Override
                            protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
                                log.debug("h2 收到的消息：{}", msg);
                                msg += "aaa";
                                ctx.fireChannelRead(msg);
                            }
                        });

                        pipeline.addLast("h3", new SimpleChannelInboundHandler<String>() {
                            @Override
                            protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
                                log.debug("h3 收到的消息：{}", msg);

                                ch.writeAndFlush(msg);
                            }
                        });

                        // 添加出站处理器
                        pipeline.addLast("o1", new ChannelOutboundHandlerAdapter() {
                            @Override
                            public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
                                log.debug("o1...");
                                super.write(ctx, msg, promise);
                            }
                        });
                        pipeline.addLast("o2", new ChannelOutboundHandlerAdapter() {
                            @Override
                            public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
                                log.debug("o2...");
                                super.write(ctx, msg, promise);
                            }
                        });
                        pipeline.addLast("o3", new ChannelOutboundHandlerAdapter() {
                            @Override
                            public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
                                log.debug("o3...");
                                super.write(ctx, msg, promise);
                            }
                        });
                    }
                })
                .bind(1122);
    }
}
